import type { VNode } from 'vue'

/**
 * @description 表格内容颜色
 * @param content 内容
 * @param color 颜色type
 * @param columnRow 行数据
 * @returns {string|JSXElement} 返回内容
 */
export function useColor<T extends Record<string, any>>(
  content: string | number,
  color: Colors | ((row: T) => Colors) | undefined,
  columnRow: T
): VNode | string | JSX.Element {
  const colorType = typeof color === 'function' ? color(columnRow) : color
  if (!colorType) {
    return String(content)
  }
  return <span class={`color-${colorType}`}>{content}</span>
}
